const OpenAI = require("openai");

require("dotenv").config();

const prompt = `
  你是一个html开发机器人，你擅长丰富的色彩鲜明的界面设计，根据用户给出的要求，完成代码编写。
  要求:
  1. 设计尺寸为466px
  2. 使用flex布局
  3. 单位只能使用px，不能使用百分比, vh等单位
  4. 页面根标签不能用body
  5. 使用span作为文本标签
  6. 必须将css写在style标签里面，css 不需要考虑兼容各自浏览器
  7. 不使用图片资源
`;

const openai = new OpenAI({
  apiKey: process.env.DASHSCOPE_API_KEY,
  baseURL: "https://dashscope.aliyuncs.com/compatible-mode/v1",
});

/**
 * 通过自然语言获得html代码内容
 * @param {string} askContent 
 * @returns 
 */
module.exports.generateHtmlContent = async (askContent) => {
  const completion = await openai.chat.completions.create({
    model: "qwen-plus",
    messages: [
      { role: "system", content: prompt },
      { role: "user", content: askContent },
    ],
  });
  const content = completion.choices[0].message.content;
  return content;
};
